import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6])
a_3=np.where(arr >= 3)
a=arr[a_3]
print("大于3的数",a)

b_2=np.where(arr < 3)
b=arr[b_2]
print("小于3的数",b)
print("将所有大于等于3的置为1:", np.where(arr > 3, arr, 1))
print("将所在小于3的成员置0:", np.where(arr < 3, arr, 0))
print("同时将所有大于等于3的置为1，将所在小于3的成员置0:",np.where(arr >= 3, 1, 0))
# Output:
#     大于3的数 [3 4 5 6]
#     小于3的数 [1 2]
#     将所有大于等于3的置为1: [1 1 1 4 5 6]
#     将所在小于3的成员置0: [1 2 0 0 0 0]
#     同时将所有大于等于3的置为1，将所在小于3的成员置0: [0 0 1 1 1 1]